import Footer from '@/components/Footer'
import RightContent from '@/components/RightContent'
import {globalModel} from '@/models/globalModel'
import {fetchUserInfo, logout} from '@/services/admin/auth'
import type {Settings as LayoutSettings} from '@ant-design/pro-components'
import {PageLoading, SettingDrawer} from '@ant-design/pro-components'
import {ApolloClient, ApolloLink, ApolloProvider, InMemoryCache} from '@apollo/client'
import type {RunTimeLayoutConfig} from '@umijs/max'
import {history} from '@umijs/max'
import defaultSettings from '../config/defaultSettings'
import {errorConfig} from './requestErrorConfig'
import {ADMIN_API} from './services/admin/typings'
import {client} from "@/global";
import {setContext} from "@apollo/client/link/context";

const isDev = process.env.NODE_ENV === 'development'
const loginPath = '/user/login'


/**
 * @see  https://umijs.org/zh-CN/plugins/plugin-initial-state
 * */
export async function getInitialState(): Promise<{
    settings?: Partial<LayoutSettings>
    currentUser?: API.CurrentUser
    userinfo?: ADMIN_API.UserInfoVO
    loading?: boolean
    fetchUserInfo?: () => Promise<API.CurrentUser | undefined>
    queryUserInfo?: () => Promise<ADMIN_API.UserInfoVO | undefined>
}> {
    const queryUserInfo = async () => {
        try {
            let userInfo = await fetchUserInfo()
            globalModel.setTenantId(userInfo.tenantInfo.id)
            globalModel.setUserId(userInfo.id)
            const authLink = setContext((_, {headers}) => {
                const token = localStorage.getItem('token');
                return {
                    headers: {
                        ...headers,
                        tenantId: userInfo.tenantInfo.id || "0"
                    },
                };
            })
            client.setLink(ApolloLink.from([authLink, client.link]))
            return userInfo
        } catch (error) {
            history.push(loginPath)
        }
        return undefined
    }

    // 如果不是登录页面，执行
    const {location} = history
    if (location.pathname !== loginPath) {
        const userinfo = await queryUserInfo()
        return {
            queryUserInfo,
            userinfo,
            settings: defaultSettings as Partial<LayoutSettings>,
        }
    }
    return {
        queryUserInfo,
        settings: defaultSettings as Partial<LayoutSettings>,
    }
}

// ProLayout 支持的api https://procomponents.ant.design/components/layout
export const layout: RunTimeLayoutConfig = ({initialState, setInitialState}) => {

    return {
        rightContentRender: () => <RightContent/>,
        waterMarkProps: {
            content: initialState?.userinfo?.userName,
        },
        logout: () => logout(),
        footerRender: () => <Footer/>,
        onPageChange: () => {
            const {location} = history
            // 如果没有登录，重定向到 login
            if (!initialState?.userinfo && location.pathname !== loginPath) {
                history.push(loginPath)
            }
        },
        layoutBgImgList: [
            {
                src: 'https://mdn.alipayobjects.com/yuyan_qk0oxh/afts/img/D2LWSqNny4sAAAAAAAAAAAAAFl94AQBr',
                left: 85,
                bottom: 100,
                height: '303px',
            },
            {
                src: 'https://mdn.alipayobjects.com/yuyan_qk0oxh/afts/img/C2TWRpJpiC0AAAAAAAAAAAAAFl94AQBr',
                bottom: -68,
                right: -45,
                height: '303px',
            },
            {
                src: 'https://mdn.alipayobjects.com/yuyan_qk0oxh/afts/img/F6vSTbj8KpYAAAAAAAAAAAAAFl94AQBr',
                bottom: 0,
                left: 0,
                width: '331px',
            },
        ],
        dynamicImport: {},
        links: [],
        menuHeaderRender: undefined,
        // 自定义 403 页面
        // unAccessible: <div>unAccessible</div>,
        // 增加一个 loading 的状态
        childrenRender: (children) => {
            if (initialState?.loading) return <PageLoading/>;
            return (
                <>
                    <ApolloProvider client={client}>
                        {children}
                        <SettingDrawer
                            disableUrlParams
                            enableDarkTheme
                            settings={initialState?.settings}
                            onSettingChange={(settings) => {
                                setInitialState((preInitialState) => ({
                                    ...preInitialState,
                                    settings,
                                })).then()
                            }}
                        />
                    </ApolloProvider>
                </>
            )
        },
        ...initialState?.settings,
    }
}

/**
 * @name request 配置，可以配置错误处理
 * 它基于 axios 和 ahooks 的 useRequest 提供了一套统一的网络请求和错误处理方案。
 * @doc https://umijs.org/docs/max/request#配置
 */
export const request = {
    ...errorConfig,
}
